#!/usr/bin/env python
"""
添加产品线示例数据
"""

import os
import django

# 设置Django环境
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'config.settings')
django.setup()

from projects.models import ProductLine

# 定义产品线数据
sample_product_lines = [
    {
        'name': '企业应用',
        'description': '适用于企业内部管理系统、业务流程管理等企业级应用'
    },
    {
        'name': '移动应用',
        'description': 'iOS、Android等移动平台应用程序'
    },
    {
        'name': '云服务',
        'description': '基于云计算平台的各种服务和应用'
    },
    {
        'name': '人工智能',
        'description': '包含机器学习、深度学习、自然语言处理等技术的应用'
    },
    {
        'name': '大数据',
        'description': '数据挖掘、分析、可视化等大数据相关应用'
    }
]

def add_product_lines():
    """添加产品线数据"""
    print('开始添加产品线数据...')
    
    for line_data in sample_product_lines:
        # 检查产品线是否已存在
        existing_line = ProductLine.objects.filter(name=line_data['name']).first()
        if existing_line:
            print(f'产品线 "{line_data["name"]}" 已存在，跳过')
            continue
        
        # 创建新产品线
        line = ProductLine.objects.create(
            name=line_data['name'],
            description=line_data['description']
        )
        print(f'成功添加产品线: {line.name}')
    
    print('产品线数据添加完成！')

if __name__ == '__main__':
    add_product_lines()